include_directories(${CMAKE_CURRENT_SOURCE_DIR}/)
file(GLOB_RECURSE CUDA_SRC ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cu)
file(GLOB_RECURSE CPU_SRC ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cc)
list(APPEND SOURCES ${CUDA_SRC} ${CPU_SRC})
# message(STATUS "====> ${CUDA_SRC} cuda source ${CPU_SRC}")
add_library(ops SHARED ${SOURCES})
